3d surface area hackerrank python